# 封装被测试系统接口
class loginAPI:
    # 初始化
    def __init__(self):
        self.login_verify_code = "http://172.16.18.128/index.php?m=Home&c=User&a=verify"
        self.login =  "http://172.16.18.128/index.php?m=Home&c=User&a=do_login"

    # 获取验证码接口
    def get_login_verify(self,session):
        # 为了获取每次接口的session，通过传入的方式请求参数
        return session.get(self.login_verify_code)

    # 登录接口
    def get_login(self,session,username,password,verify_code):
        login_data = {
            "username": username,
            "password": password,
            "verify_code": verify_code
        }
        return session.post(self.login,data=login_data)
